package com.yifeng.util;

import com.yifeng.po.User;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.subject.Subject;

/**
 * 用户上下文，和用户相关的数据都可以放到里面 比如：登录用户信息，用户拥有的权限
 *
 * @author Administrator
 *
 */
public class UserContext {

	private static final String LOGINUSER = "loginUser";

	/**
	 * 把当前用户存入Session
	 *
	 * @param loginUser
	 */
	public static void setUser(User loginUser) {

		// 获取当前主体
		Subject currentUser = SecurityUtils.getSubject();
		// 把用户放入Session
		currentUser.getSession().setAttribute(LOGINUSER, loginUser);
	}

	public static User getUser() {
		// 获取当前主体
		Subject currentUser = SecurityUtils.getSubject();
		// 把用户放入Session
		return (User) currentUser.getSession().getAttribute(LOGINUSER);
	}
}
